#include "blockqueue.hpp"
#include "Task.hpp"
#include <unistd.h>
#include <ctime>

void* Comsumer(void* args)
{
    BlockQueue<Task>* bq = static_cast<BlockQueue<Task>*>(args);

    while (true)
    {
        Task t = bq->pop(); // 消费

        t(); // 计算

        std::cout << std::cout << "处理任务: " << t.GetTask() << " 运算结果是： " 
        << t.GetResult() << " thread id: " << pthread_self() << std::endl;

        sleep(1);
    }
}

void* Productor(void* args)
{
    BlockQueue<Task>* bq = static_cast<BlockQueue<Task>*>(args);
    int len = opers.size();

    while (true)
    {
        int data1 = rand() % 10 + 1;
        usleep(10); // 随机性更大一些
        int data2 = rand() % 10;
        char op = opers[rand() % len];

        Task t(data1, data2, op);

        bq->push(t);
        std::cout << "生产了一个任务: " << t.GetTask() << " thread id: " 
        << pthread_self() << std::endl;
    }
}

int main()
{
    srand(time(nullptr));

    BlockQueue<Task>* bq = new BlockQueue<Task>();
    pthread_t c[3], p[5];

    for (int i = 0; i < 3; i++)
    {
        pthread_create(c + i, nullptr, Comsumer, bq);
    }

    for (int i = 0; i < 5; i++)
    {
        pthread_create(p + i, nullptr, Productor, bq);
    }

    for (int i = 0; i < 3; i++)
    {
        pthread_join(c[i], nullptr);
    }

    for (int i = 0; i < 5; i++)
    {
        pthread_join(p[i], nullptr);
    }

    delete bq;

    return 0;
}